package com.nj.dms.util;

/**
 * 获取文件类型
 * 
 * @author Administrator
 *
 */
public class FileType {

	public static int getFileType(String fileName) {
		if (fileName != null) {
			String fileType = "";
			try {
				// 获取文件后缀名并转化为小写，用于后续比较
				fileType = fileName.substring(fileName.lastIndexOf("."), fileName.length()).toLowerCase();

			} catch (StringIndexOutOfBoundsException e) {
				e.getMessage();
			}

			// 创建图片类型数组
			String img[] = { ".bmp", ".jpg", ".jpeg", ".png", ".tiff", ".gif", ".pcx", ".tga", ".exif", ".fpx", ".svg",
					".psd", ".cdr", ".pcd", ".dxf", ".ufo", ".eps", ".ai", ".raw", ".wmf" };
			for (int i = 0; i < img.length; i++) {
				if (img[i].equals(fileType)) {
					return 2;
				}
			}

			// 创建文档类型数组
			String document[] = { ".txt", ".doc", ".docx", ".xls", ".htm", ".html", ".jsp", ".rtf", ".wpd", ".pdf",
					".ppt" };
			for (int i = 0; i < document.length; i++) {
				if (document[i].equals(fileType)) {
					return 3;
				}
			}
			// 创建视频类型数组
			String video[] = { ".mp4", ".avi", ".mov", ".wmv", ".asf", ".navi", ".3gp", ".mkv", ".f4v", ".rmvb",
					".webm" };
			for (int i = 0; i < video.length; i++) {
				if (video[i].equals(fileType)) {
					return 4;
				}
			}
			// 创建音乐类型数组
			String music[] = { ".mp3", ".wma", ".wav", ".mod", ".ra", ".cd", ".md", ".asf", ".aac", ".vqf", ".ape",
					".mid", ".ogg", ".m4a", ".vqf" };
			for (int i = 0; i < music.length; i++) {
				if (music[i].equals(fileType)) {
					return 5;
				}
			}

		} else {

			// fileName = "文件名为空！";
			return 0;

		}
		return 1;
	}
}